package com.atguigu.gmall.common.config;

import com.alibaba.fastjson.JSON;
import com.atguigu.gmall.common.model.GmallCorrelationData;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.concurrent.TimeUnit;

@Configuration
@Component
@Slf4j
public class MQProducerAckConfig implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    //  初始化，套用这个磨板！
    @PostConstruct
    public void init(){
        //  this 表示当前这个类对象
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 判断消息是否到交换机
     * @param correlationData
     * @param ack
     * @param cause
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack){
            log.info("消息到交换机");
        }else {
            log.error("消息未到交换机"+cause);
            //  调用重试方法。
            this.retrySendMsg(correlationData);
        }
    }


    /**
     * 当消息没有到队列中才会执行！
     * @param message
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息路由queue失败，应答码={}，原因={}，交换机={}，路由键={}，消息={}",
                replyCode, replyText, exchange, routingKey, message.toString());

        //  if("延迟") return;
        //  只要能走到这个地方，就说明消息没有到队列中！ 获取缓存中的数据！
        //  获取缓存的key
        String key = (String) message.getMessageProperties().getHeaders().get("spring_returned_message_correlation");
        String jsonStr = (String) this.redisTemplate.opsForValue().get(key);
        //  将数据转换为GmallCorrelationData 对象
        GmallCorrelationData gmallCorrelationData = JSON.parseObject(jsonStr, GmallCorrelationData.class);
        //  判断延迟
        if (gmallCorrelationData.isDelay()) return;
        //  调用重试方法
        this.retrySendMsg(gmallCorrelationData);
    }

    /**
     * 重试方法
     * @param correlationData
     */
    private void retrySendMsg(CorrelationData correlationData) {
        //  先获取重试次数
        GmallCorrelationData gmallCorrelationData = (GmallCorrelationData)correlationData;
        int retryCount = gmallCorrelationData.getRetryCount();
        //  判断重试次数
        if (retryCount>=3){
            log.error("消息重试次数已到 {}"+retryCount);
            //  insert into msg value(id,exchange,routingKey,msg);  有人解决这个问题！
        } else {
            //  更新重试次数
            retryCount++;
            gmallCorrelationData.setRetryCount(retryCount);
            //  写回缓存！
            redisTemplate.opsForValue().set(gmallCorrelationData.getId(), JSON.toJSONString(gmallCorrelationData),10, TimeUnit.MINUTES);
            //  判断
            if (gmallCorrelationData.isDelay()){
                //  发送延迟消息
                rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),
                        (message)->{
                            //  设置延迟时间：10s
                            message.getMessageProperties().setDelay(1000*gmallCorrelationData.getDelayTime());
                            return message;
                        },gmallCorrelationData);
            } else {
                //  重发普通消息！
                rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),gmallCorrelationData);
            }
        }
    }
}
